# Copyright (c) 2012, Suraj Gowda 
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of the <organization> nor the
#       names of its contributors may be used to endorse or promote products
#       derived from this software without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL Suraj Gowda BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

from numpy import *
from mlabwrap import mlab
import mlabraw, re


def is_cell_proxy(obj):
    return "<MlabObjectProxy of matlab-class: 'cell'" in str(obj)

def is_struct_proxy(obj):
    return "<MlabObjectProxy of matlab-class: 'struct'" in str(obj)

def convert(obj):
    if isinstance(obj, basestring):
        return obj
    elif isinstance(obj, ndarray):
        if shape(obj) == (1,1):
            return obj[0,0]
        else:
            return obj
    elif is_cell_proxy( obj ):
        m = re.match(".*?internal name: '(.*?)';.*?", str(obj))
        varname = m.group(1)
        arr_len = int( mlab.eval('length(%s)' % varname)[0,0] )

        cell_list = []
        for k in range(arr_len):
            matlab_idx = k+1
            cell_list.append( convert( mlab.eval('%s{%d}' % (varname, matlab_idx)) ))
        return cell_list            

    elif is_struct_proxy(obj):
        m = re.match(".*?internal name: '(.*?)';.*?", str(obj))
        varname = m.group(1)
        fieldnames = convert(mlab.fieldnames(obj))
        dict = {}
        for key in fieldnames:
            dict[key] = convert(mlab.eval('%s.(\'%s\')' % (varname, key) ))
    
        return dict
        
        

